from matplotlib import pyplot
import numpy

import phaseportrait

from pyscript import web, when, display

global alpha_1
global alpha_2

global beta_1
global beta_2

global gamma_1
global gamma_2

global A_1
global A_2

def love_dynamics(x, y):
	global alpha_1
	global alpha_2

	global beta_1
	global beta_2

	global gamma_1
	global gamma_2

	global A_1
	global A_2

	return -alpha_1*x + beta_1*y + gamma_1*A_2, -alpha_2*y + beta_2*x + gamma_2*A_1

@when("click", "#graph-button")
def main(event):
	global alpha_1
	global alpha_2

	global beta_1
	global beta_2

	global gamma_1
	global gamma_2

	global A_1
	global A_2

	print("alpha_1 value:")
	alpha_1 = float( web.page["alpha_1"].value )
	print("alpha_2 value:")
	alpha_2 = float( web.page["alpha_2"].value )

	print("beta_1 value:")
	beta_1 = float( web.page["beta_1"].value )
	print("beta_2 value:")
	beta_2 = float( web.page["beta_2"].value )

	print("gamma_1 value:")
	gamma_1 = float( web.page["gamma_1"].value )
	print("gamma_2 value:")
	gamma_2 = float( web.page["gamma_2"].value )

	print("A_1 value:")
	A_1 = float( web.page["A_1"].value )
	print("A_2 value:")
	A_2 = float( web.page["A_2"].value )

	SimplePendulum = phaseportrait.PhasePortrait2D(love_dynamics, [0, 100], Title='Love Dynamics', xlabel="Love of Person 1", ylabel="Love of Person 2")
	figure, axes = SimplePendulum.plot()

	display(figure, append=False)